package com.suse.generator.mybatis.plus;

import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.po.LikeTable;

public class StrategyConfigBuilder {

    private final StrategyConfig sc;

    public StrategyConfigBuilder(StrategyConfig sc) {
        this.sc = sc;
    }

    public StrategyConfigBuilder tablePrefix(String prefix) {
        sc.setTablePrefix(prefix);
        return this;
    }

    public StrategyConfigBuilder likeTable(String like) {
        sc.setLikeTable(new LikeTable(like));
        return this;
    }

    public StrategyConfigBuilder include(String... table) {
        sc.setInclude(table);
        return this;
    }

}
